import numpy as np

# 创建示例数组
arr = np.array(
    [
        [1, 2, 3],
        [4, 5, 6],
        [7, 8, 9],
    ]
)

# 查找 >5 的元素
row, col = np.where(arr > 5)
for i in range(len(row)):
    print(f"位置row:{row[i]} col:{col[i]}, {arr[row[i]][col[i]]} > 5")
"""
位置row:1 col:2, 6 > 5
位置row:2 col:0, 7 > 5
位置row:2 col:1, 8 > 5
位置row:2 col:2, 9 > 5
"""

# 查找 3 <= x <= 5 的元素
row, col = np.where((arr >= 3) & (arr < 5))
for i in range(len(row)):
    print(f"位置row:{row[i]} col:{col[i]}, 3 <= {arr[row[i]][col[i]]} < 5")
"""
位置row:0 col:2, 3 <= 3 < 5
位置row:1 col:0, 3 <= 4 < 5
"""
